function [Train, Test] = getTrainTestFiles(net_output, Train, Test)

data_path = srpath.getDataPath;

switch net_output
    case 'r'
        Test.files  = fullfile(data_path, 'testres.h5');
    case 'y'
        Test.files  = fullfile(data_path, 'test.h5');
end

switch Train.batch_size
    case 64
        switch net_output
            case 'r'
                Train.files = fullfile(data_path, 'train291-aug234stdr41res.h5');
            case 'y'
                Train.files = fullfile(data_path, 'train291-aug234stdr41.h5');
        end
    case 128
        switch net_output
            case 'r'
                Train.files = fullfile(data_path, 'train291-aug234stdr41res_128.h5');
            case 'y'
                Train.files = fullfile(data_path, 'train291-aug234stdr41_128.h5');
        end
end
